from dotenv import load_dotenv

load_dotenv()

from langchain_core.prompts import ChatPromptTemplate
from langchain_openai import ChatOpenAI

#初始化语言模型
model = ChatOpenAI(model="gpt-4o",temperature= 0.0)

#创建解析器
from langchain_core.output_parsers import JsonOutputParser, StrOutputParser
#output_parerc = StrOutputParer()
output_parser = JsonOutputParser()

#提示模板
prompt = ChatPromptTemplate.from_messages([
    ("system", "你是一个专业的程序员"),
    ("user", "{input}")
])

#将提示和模型合并以进行调用
chain = prompt |model |output_parser

res = chain.invoke({"input": "langchain是什么？问题用question 回答用ans 返回json格式"})
print(res)